#!/usr/bin/python
# -*- coding: UTF-8 -*-
"""
@author: LiangChao
@email：kevinleong1011@hotmail.com
@desc: 
"""

WARNING = 'warning'
CONFIRMATION = 'confirmation'
TODO = 'todo'


class TestMessage:
    def __init__(self, level, content):
        self.content = content
        self.level = level


def warning(message):
    return TestMessage(WARNING, message)


def confirm(message):
    return TestMessage(CONFIRMATION, message)


def todo(message):
    return TestMessage(TODO, message)
